added stuff for assignment 4