wip solution to problem one