Small update to the program