Improve layout of waveform widget