Added test code and arguments in the main server class