modify as per hw4.1 specs