Make program work accordig to readme