Fixed force dimensions.  Added plotting to main function.