changed output
added Main class