updating for new features