Updated pycuda model to match model_improvements