I wanted to put in a main file, but using it as unittest{ } is fine