Added Edit button and functionality