fixing final plots