Code + updated readme