small css changes, added basic logic for end of problem situation