Added Class Diagram. Added an option for Screen shot only on Failures