Necessary changes to the program