Assignment 1 - almost done except by one test