added some code
need to fix original color