Small changes

Move legend downscreen and fix up a coment.