assignment 5 code finished