DAMI assignment 2