added some stuff to main