Improved print statements and updates readme