completed assignemnt and added test