Added if else, changed readme