diff --git a/R/ageRatioTest.R b/R/ageRatioTest.R
index 08d6832..f7a16b6 100644
--- a/R/ageRatioTest.R
+++ b/R/ageRatioTest.R
@@ -2,12 +2,18 @@
 #
 #' Age ratio test
 #'
-#' Age Ratio Test is an age-related test of survey and data quality.
+#' Age ratio test is an age-related test of survey and data quality. In this
+#' test, the ratio of the number of children aged from 6 to 29 months to the
+#' number of children aged from 30 to 59 months is calculated. This ratio is
+#' then compared to an expected ratio (usually set at 0.85). The difference
+#' of the observed ratio to the expected ratio is then compared statistically
+#' using Chi-squared test.
 #'
-#' @param x Numeric vector (age)
-#' @param ratio Expected age ratio
+#' @param x A vector for age. Should either be in whole months (integer) or in
+#'   calculated decimal months (numeric).
+#' @param ratio Expected age ratio. Default is 0.85.
 #'
-#' @return A lit of class `"ageRatioTest"` with:
+#' @returns A lit of class `"ageRatioTest"` with:
 #'
 #' | **Variable** | **Description** |
 #' | :--- | :--- |
@@ -20,7 +26,7 @@
 #' | *p* | `p-value` for Chi-squared test |
 #'
 #' @examples
-#' # Age-ratio test on survey dataset from Kabul, Afghanistan (dp.ex02)
+#' # Age-ratio test on survey dataset from Kabul, Afghanistan (`dp.ex02`)
 #' # with an age ratio of 0.85
 #' svy <- dp.ex02
 #' ageRatioTest(svy$age, ratio = 0.85)
@@ -34,11 +40,22 @@
 ################################################################################
 
 ageRatioTest <- function(x, ratio = 0.85) {
-  g <- recode(x, "6:29=1; 30:59=2")
+  ## If x is numeric ----
+  if (is.numeric(x)) x <- floor(x)
+
+  ## If x is integer ----
+  if (is.integer(x)) x <- x
+
+  ## If x is not numeric or integer ----
+  if (!is.numeric(x) & !is.integer(x))
+    stop("Age should be of class integer or numeric. Try again.")
+
+  ## Calculate age ratio ----
+  g <- ifelse(x < 30, 1, 2)
   expectedP <- ratio / (ratio + 1)
-  observedP <- sum(g == 1)/ sum(table(g))
+  observedP <- sum(g == 1, na.rm = TRUE)/ sum(table(g))
   observedR <- observedP / (1 - observedP)
-  X2 <- prop.test(sum(g == 1), sum(table(g)), p = expectedP)
+  X2 <- prop.test(sum(g == 1, na.rm = TRUE), sum(table(g)), p = expectedP)
   result <- list(expectedR = ratio,
                  expectedP = expectedP,
                  observedR = observedR,
@@ -61,7 +78,7 @@ ageRatioTest <- function(x, ratio = 0.85) {
 #' @return Printed output of [ageRatioTest()] function
 #'
 #' @examples
-#' # Print age-ratio test results for survey dataset from Kabul, Afghanistan (dp.ex02)
+#' # Print age-ratio test results for survey dataset from Kabul, Afghanistan
 #' svy <- dp.ex02
 #' print(ageRatioTest(svy$age, ratio = 0.85))
 #'
diff --git a/man/ageRatioTest.Rd b/man/ageRatioTest.Rd
index 92160fd..ef00e87 100644
--- a/man/ageRatioTest.Rd
+++ b/man/ageRatioTest.Rd
@@ -7,9 +7,10 @@
 ageRatioTest(x, ratio = 0.85)
 }
 \arguments{
-\item{x}{Numeric vector (age)}
+\item{x}{A vector for age. Should either be in whole months (integer) or in
+calculated decimal months (numeric).}
 
-\item{ratio}{Expected age ratio}
+\item{ratio}{Expected age ratio. Default is 0.85.}
 }
 \value{
 A lit of class \code{"ageRatioTest"} with:\tabular{ll}{
@@ -24,10 +25,15 @@ A lit of class \code{"ageRatioTest"} with:\tabular{ll}{
 }
 }
 \description{
-Age Ratio Test is an age-related test of survey and data quality.
+Age ratio test is an age-related test of survey and data quality. In this
+test, the ratio of the number of children aged from 6 to 29 months to the
+number of children aged from 30 to 59 months is calculated. This ratio is
+then compared to an expected ratio (usually set at 0.85). The difference
+of the observed ratio to the expected ratio is then compared statistically
+using Chi-squared test.
 }
 \examples{
-# Age-ratio test on survey dataset from Kabul, Afghanistan (dp.ex02)
+# Age-ratio test on survey dataset from Kabul, Afghanistan (`dp.ex02`)
 # with an age ratio of 0.85
 svy <- dp.ex02
 ageRatioTest(svy$age, ratio = 0.85)
diff --git a/man/print.ageRatioTest.Rd b/man/print.ageRatioTest.Rd
index 6f0b23e..6959ac9 100644
--- a/man/print.ageRatioTest.Rd
+++ b/man/print.ageRatioTest.Rd
@@ -18,7 +18,7 @@ Printed output of \code{\link[=ageRatioTest]{ageRatioTest()}} function
 \code{\link[=print]{print()}} helper function for \code{\link[=ageRatioTest]{ageRatioTest()}} function
 }
 \examples{
-# Print age-ratio test results for survey dataset from Kabul, Afghanistan (dp.ex02)
+# Print age-ratio test results for survey dataset from Kabul, Afghanistan
 svy <- dp.ex02
 print(ageRatioTest(svy$age, ratio = 0.85))
 
diff --git a/tests/testthat/test_ageRatioTest.R b/tests/testthat/test_ageRatioTest.R
index 51cf9b1..afe222e 100644
--- a/tests/testthat/test_ageRatioTest.R
+++ b/tests/testthat/test_ageRatioTest.R
@@ -25,3 +25,73 @@ test_that("names of art", {
 test_that("print(art) message exists", {
   expect_output(print(art))
 })
+
+## Test that function works when age has NAs
+
+svy <- dp.ex02
+
+svy$age[sample(seq_len(length(svy$age)), 5)] <- NA_integer_
+
+art <- ageRatioTest(svy$age, ratio = 0.85)
+
+test_that("art is ageRatioTest", {
+  expect_is(art, "ageRatioTest")
+})
+
+test_that("art is list", {
+  expect_true(is.list(art))
+})
+
+test_that("names of art", {
+  expect_equal(names(art)[1], "expectedR")
+  expect_equal(names(art)[2], "expectedP")
+  expect_equal(names(art)[3], "observedR")
+  expect_equal(names(art)[4], "observedP")
+  expect_equal(names(art)[5], "X2")
+  expect_equal(names(art)[6], "df")
+  expect_equal(names(art)[7], "p")
+})
+
+test_that("print(art) message exists", {
+  expect_output(print(art))
+})
+
+## Test that function works when age is numeric value
+
+svy <- dp.ex02
+
+svy$age <- as.numeric(svy$age)
+
+art <- ageRatioTest(svy$age, ratio = 0.85)
+
+test_that("art is ageRatioTest", {
+  expect_is(art, "ageRatioTest")
+})
+
+test_that("art is list", {
+  expect_true(is.list(art))
+})
+
+test_that("names of art", {
+  expect_equal(names(art)[1], "expectedR")
+  expect_equal(names(art)[2], "expectedP")
+  expect_equal(names(art)[3], "observedR")
+  expect_equal(names(art)[4], "observedP")
+  expect_equal(names(art)[5], "X2")
+  expect_equal(names(art)[6], "df")
+  expect_equal(names(art)[7], "p")
+})
+
+test_that("print(art) message exists", {
+  expect_output(print(art))
+})
+
+
+## Test that function works when age is character value
+
+svy <- dp.ex02
+
+svy$age <- as.character(svy$age)
+
+expect_error(ageRatioTest(svy$age))
+