Improve amplitude plots