Move return class after class definition. Fixes PHP 5.5 fatal errors.

This is needed otherwise it will result in a fatal error for PHP 5.5