Finished the first two problems