updated main so code compiles and runs