Created class for selected units, Fixes #62