Added styling for figure