Added first version of a trait selection box