Minor changes to problem