Added simple extension to work off