made the code look nice