fixing layout

changing wrapper a bit