Added some stuff in main