Fixed units
Merge pull request #6 from anantmittal/master

Fixed units