optimise first part of the theta step

Signed-off-by: Mattias Andrée <maandree@operamail.com>