Add invitation model
Small fixes