TODO: fix hitboxes