Main changed for test