adding better scaling for models