Adding funcatioalty