remodelled einstein with correct solution