Edit function WORKS!!

Figured out how to get the edit button working! Also updated comments.