Added selection widget to Unit