Quick fix by Steven Love
Merge pull request #7 from chrismullins/fix-for-nonsquare-wave

Quick fix by Steven Love