Make Units class bit more pythonic