slight design updates