Add print function
Update main