Add some functionality