Refactored __init__.py
Add shooting logic