Add simple entity AI and knockback