layout/bugs fixes