fix code and finish 1b)