Changes to logic has been made.

1. Print statements removed.
2. returns added