Adds code and updates readme