DST assignment 1 including tests