Add unittests. Remove specs
Add post_initialize method.