Add test and refactor 'unlock_levels'