Changed the main script to test both methods
Updated readme for the problem 2