Updates for styling issues