Making a take function for easier lazy list viewing