StatML assignment 1