Move selection box info into class