Added changes to program #1