exercise 7 - clean & fix code