adding AI rules. TODO: Make less messy and smarter