switched to cout/cin
Added options for the UI