more bug fixes.!