Finished the shell code for assignment 2