assig2 modified