Add scale function