changed main to work! (hopefully)