Adding methods and small amount of functionality