Lab4 bug resolved