While running through questions for assessment, an optional button to return to home screen or option saying - "I will do it later" can be added.

Currently, Save as Draft provides good flexibility to return later, but for that, user needs to click on cancel to return to main screen.

This additional option can enhance usability.