Added edit post button and controller