Add support for scaling