finishing up assignment 4