started working again with assignment1