[Axis] Fixed some mouse behavior, refactored