refactoring / dealing with args
refactoring args input