Added test and code for color module