Making 'Problem' button work. Fixes #2