[TASK] Move frontend code back into extension