Fixing stuff in homework 0