add arguments

tweaking, add arguments code