Cleaned up code in LFO test