Simpler program that works!