Finished problem 1. YAY!