Made check work properly. Added colors.