fixed display logic and did a little css