Add initial version of extension + unit tests