Removed UIColor and replaced with self