working towards fixing the CSS / styling