program working, need to work on AI