Fixed Ben's issues

Removed test code
Added app.py