Add csrf logic