Homework for 222 done, could make prettier