Yep, but main thing - separate IF/ELSE from main logic, and don't hesitate to write helper functions for yourself to make IF/ELSE/definitions/error handling easier.

Also, test your logic with unit tests to prevent regressions!