design fixes

design