updated menu screen code a bit