Add some functionallity