src: added proper end screen