Added missing class and screen capture