made my code look neater