Version 2.1 app.py

Fixed 0 and 1 issue with prime numbers.