adding further changes to Project 4 main file