Add surface function plotting