Made my code look better

Thankyou John