Fixed the size of the edit box while editing it. It has some jumping and slight marging issue, but at least it works