Adding rest of the changes by Kristina. Some minor improvements and improved dataset chapter.