modified Assignment_3