Updated readme + fixed code