Add extension initailization (#321)