update css to use one container div
create wrapper function