changed class name to make it work