Improved AI logic and attak animation