Fix the program so i work