Color Preservation

Added some functionality to the code highlighting in an attempt at
preserving the code color heirarchy.