assignment 4 bugs corrections