Proeprly passing use case argument
Updated demo