todo. bug fixes.