@wip Problem 1 finished