Place function into main method
Add alternative capture method using pygame