Added: Began with return statement implementation (not finished yet)