Got the program to work