[TASK] Implement correct assertion step validation
[TASK] Remove assertions, better error messages, fix styling