Made code nicer :D