modified MainTest again