ENH: Cleaned source code
Added setModelColor method