Add argument passing to tests