Added enum for direction