Corrected dimWeight to correct dimension
Added small test for momentumTask