code fixed, working display