* Implement surf()
* Implement plot using matplotlib
* Change colour images to be H x W x 3, instead of 3 x H x W. Be consistent with matplotlib
