modify map method of OHM class