FIX : UI Changes