Add problem id

@arthurnn