Adding functionlaity