UI issue fixed.
Removed hardcoded message from Edit class.