still trying to fix omptimzation