Move mutation observer out of init and UI/code clean up