1 files changed, 9 insertions, 1 deletions

diff --git a/main.cpp b/main.cpp
index b6e0a73..ecfd19d 100644
--- a/main.cpp
+++ b/main.cpp
@@ -46,7 +46,6 @@ int main(int argc, char* args[]) {
                         break;
                     case SDLK_UP:
                     case SDLK_w:
-                    case SDLK_SPACE:
                         level.player.jump();
                         break;
                     case SDLK_LCTRL:
@@ -57,6 +56,15 @@ int main(int argc, char* args[]) {
                         if (level.player.getCrouch() == 0)
                             level.player.setCrouch(2);
                         break;
+                    case SDLK_SPACE:
+                        level.player.attack();
+                        break;
+                    case SDLK_0:
+                        level.player.setWeapon(0);
+                        break;
+                    case SDLK_1:
+                        level.player.setWeapon(1);
+                        break;
                 }
             } else if (event.type == SDL_KEYUP) {
                 // Adjust player velocity