Fix checking before and after done work.
Merge branch 'master' of github.com:TTK4145-Project/TTK4145-Project