updating for assignment 5