Adding tests at the end