[TASK] Implements most of the needed functionality for the extension