Update main.cs

Added device enum