Making figure nicer