Bte2601 assignment 4