Added mods back to lib.rs to actually run the tests