updated with css fix