hw1 restored main to work with thrust