Fixed some methods, working on Medium AI level