Input Added to main.cpp