Your proposed modifications sound good. In your final submission be sure to describe how each level you provide is unique from the others so we know what to look for when grading.