added edit funcationality