Add inputs and outputs to python models