Improvement in editmode

added visual improvent for when in editmode