added some code to almost finish it
Finished the program