working on fixing initial program