made the code look better