Style fixes for extension class, removed unneeded code.
Added missing tests for extension class.
Merge pull request #3 from Zylius/tests

Improvements for extension class.