Little cleanup
Little cleanup