Fixing optional arguments logic