Add the function "at". Check its funtioning in the main.