Finished with code, need to clean up and add scene