Little optimisation but I got stuck