Adding tests finally.