Add code and update README