Attempt at adding testing