And a little more logic
Solved 54 but with bad style