Minor fixes lab4 WP