Added function calls to main