Working Ai with proper weighting.. only in dimension selection