here is the rest of the code, it should work fine