Program already working

Done assignment