Write main body of program
Add test