Rewrote how you set your selection and changed it a bit. Fixes #2

Signed-off-by: Ollie <dzineit@live.co.uk>