Minor fixes for lab3