Updated Initialize class, need to find a way to get user input