fixed go-back-to-plot function which was broken along the way
changed a little bit of styling