Added in Exercise 3, dunno if it works.