Feature fixing

Screen refresh bug