[FEATURE] Improve engineering for Model command
[TASK] Beautify code and remove old todo
[FEATURE] Add flag for saving code to file system