Bug with input for Main class fixed.