Removed print statements
Added run method