Add functionality for multiple levels