Cleaned up code. Removed flask app