started tidying up equations