Test cases work but script is unfinished