Bug Fix: Missing brace to close out class