Forgot to update LAB4. Working as intended