Your proposed modifications sound good, just be sure to be descriptive in the final submission about how each new level is unique from the others and what new abilities we should be looking for when grading.