Move ai modules to ai directory. Include 2 new ai modules