Added edit button and functionality; style changes