I forgot to commit this class on last commit