Clean up styling issues