Implemented setSelection(), removed TODO
Avoided possible NPE
Fixed: setSelection(Selection selection);