Think it works now!
Solved bug in AABB