Add additional functionality