Fixing windows_in_direction function! Closes #6!