correct stupid mistake, add tests
finished challenge 1 - week04