hotfix: forgot 'end' keyword after class definition