Clean up python warnings