add readme + fix logic