added more code and more references to literature review