making the code pretty. Part 1