Tidy up
Allow dimensions of output be changed

Closes #11.