Update to allow for selection of levels