Added some extra functions. Holding off on AI portion for now.