Add an edit functional, but it looking so poor