made the code prettier