Simplified code
 Updated COL dataset