Adding tests and they passed