Simply copied over the code from last weeks assignment... (seems to have worked)