Adding functinality