edited main so program compiles