Quick edits to main function for testing purposes.