added print statements
temp
notes + refactoring + fix