Updated

To comply with the updated main.java