wrap up code and improve it. Almost ready to turn in