cleanup
added color scheme support, but still not working properly when called from an external script