Fixed screenshot and minor graphical issues