Cleaned up doc strings