Edited to write a simple program