Yeah sounds like mine is pretty normal then, thanks!
It's still a bit of a work in progress, right now you have to push the cap in to turn it on and pull it out to turn it off, I think a nice switch would be the finishing touch!
It's pretty simple though, the only part you haven't seen is...