Add friction field plotting, other bux fixes