How to scale data and keep axis correct?