Enhance H in assignment 2