Added partial tests for extension