Fix error after modification. And small block homework #15