Yet another fix to coding assignment