Adding log4j
Cleaning up code
updated color model/train file
Cleaned up test cases