problem 1

Completed! Probably did it in a stupid way though.