updating code to include first actual test