code is working, project is just about done