Project 4 initial commit; need to finish main function