added more to Assignment 4