Updated docs and tidied example