Adding Edit functationality