Add function main