Minor changes to make the program actually run