Added more functionality for future levels