Final project, IA improved a little bit