Tidy code up

Tidy code up for https://github.com/benpickles/peity/pull/43