update theme to include square code