updated tests and main function, just testing