Removed testing code from main.