few design fix