Added edit function, formatted some css