Adding setup and main