Adding happy path testing