Updated benchmark code to run potential and gz

Runs on smaller model and grid to make it more practical.