Fix plotting
Merge pull request #61 from mlolson/transfer-function

Fix plotting