Working on a mini stack implementation