sound: Add sample rate as a parameter for square wave