Made exercise 4 shorter