Updated some color scaling, working a little bit better.

Still needs work.
Added box to be used for rotation.