Refactor
Remove comments
Remove prints
Remove comments
Update docs to reflect the changes
Remove comments