Cleaned up code based on pylint recommendations.
Fixed bug where prompt function would not be launched