Finish Project 4 code