Snapshot after ML homework 5