now can handle arguments correct