ML assignment 1 - feature normalization, vectorized version