Adding register that holds the previous round key generated. This should be done anyway.
Fixed missing non blocking assignment.