[TASK] Add input message
[BUGFIX] Fix wrong calculation of sz