Added function calls to code, removed old prints