Use args to simplify the code