fixing some styling in homework