Add code and update readme