diff --git a/src/main/java/NTNU/IDATT1002/controllers/CreateAlbum.java b/src/main/java/NTNU/IDATT1002/controllers/CreateAlbum.java
index 4e4cc2b8d3539d8deea2e4f7ce834286c84c1de5..d95c13d8a47dcbbaec719663bf169e55cad28248 100644
--- a/src/main/java/NTNU/IDATT1002/controllers/CreateAlbum.java
+++ b/src/main/java/NTNU/IDATT1002/controllers/CreateAlbum.java
@@ -38,159 +38,184 @@ import java.util.stream.Collectors;
 /**
  * Controls the buttons and changeable elements on create_album.fxml,
  * a page where you create albums
+ *
  * @version 1.0 22.03.2020
  */
 public class CreateAlbum implements Initializable {
-    public TextField tbar_search;
-    public ImageView tbar_logo;
-    public Button tbar_explore;
-    public Button tbar_map;
-    public Button tbar_upload;
-    public TextField album_title_field;
-    public TextField album_tag_field;
-    public TextArea album_desc_field;
-    public Pane metadata_pane;
-    public GridPane image_grid;
-    public Button add_images_button;
-    public Button create_album_button;
-    public Button tbar_albums;
-    public Button tbar_searchBtn;
-    public VBox fileContainer;
-
-    private AlbumService albumService;
-    private ImageService imageService;
-
-    public CreateAlbum() {
-        EntityManager entityManager = App.ex.getEntityManager();
-        albumService = new AlbumService(entityManager);
-        imageService = new ImageService(entityManager);
+  public TextField tbar_search;
+  public ImageView tbar_logo;
+  public Button tbar_explore;
+  public Button tbar_map;
+  public Button tbar_upload;
+  public TextField album_title_field;
+  public TextField album_tag_field;
+  public TextArea album_desc_field;
+  public Pane metadata_pane;
+  public GridPane image_grid;
+  public Button add_images_button;
+  public Button create_album_button;
+  public Button tbar_albums;
+  public Button tbar_searchBtn;
+  public VBox fileContainer;
+
+  private AlbumService albumService;
+  private ImageService imageService;
+
+  public CreateAlbum() {
+    EntityManager entityManager = App.ex.getEntityManager();
+    albumService = new AlbumService(entityManager);
+    imageService = new ImageService(entityManager);
+  }
+
+  @Override
+  public void initialize(URL url, ResourceBundle resourceBundle) {
+    List<Image> allImages = imageService.getImageFromUser(ApplicationState.getCurrentUser());
+    for (Image image : allImages) {
+      javafx.scene.image.Image convertedImage = ImageUtil.convertToFXImage(image);
+      HBox container = new HBox();
+      container.setPrefWidth(450);
+      container.setAlignment(Pos.TOP_CENTER);
+      ImageView imageView = new ImageView();
+      imageView.setFitHeight(200);
+      imageView.setFitWidth(350);
+      imageView.setPickOnBounds(true);
+      imageView.setPreserveRatio(true);
+      imageView.setImage(convertedImage);
+      CheckBox checkBox = new CheckBox();
+      checkBox.setId(image.getId().toString());
+      container.getChildren().addAll(imageView, checkBox);
+      fileContainer.getChildren().add(container);
     }
-
-    @Override
-    public void initialize(URL url, ResourceBundle resourceBundle) {
-        List<Image> allImages = imageService.getImageFromUser(ApplicationState.getCurrentUser());
-        for (Image image : allImages){
-            javafx.scene.image.Image convertedImage = ImageUtil.convertToFXImage(image);
-            HBox container = new HBox();
-            container.setPrefWidth(450);
-            container.setAlignment(Pos.TOP_CENTER);
-            ImageView imageView = new ImageView();
-            imageView.setFitHeight(200);
-            imageView.setFitWidth(350);
-            imageView.setPickOnBounds(true);
-            imageView.setPreserveRatio(true);
-            imageView.setImage(convertedImage);
-            CheckBox checkBox = new CheckBox();
-            checkBox.setId(image.getId().toString());
-            container.getChildren().addAll(imageView, checkBox);
-            fileContainer.getChildren().add(container);
-        }
-    }
-
-    /**
-     * Method that changes stage to Main page
-     * @param mouseEvent
-     * @throws IOException
-     */
-    public void switchToMain(MouseEvent mouseEvent) throws IOException {
-        App.setRoot("main");
+  }
+
+  /**
+   * Method that changes stage to Main page
+   *
+   * @param mouseEvent
+   * @throws IOException
+   */
+  public void switchToMain(MouseEvent mouseEvent) throws IOException {
+    App.setRoot("main");
+  }
+
+  /**
+   * Method that changes stage to Search page. It reads the value of the search
+   * field and if not empty it is passed to dataexchange
+   *
+   * @param actionEvent
+   * @throws IOException
+   */
+  public void switchToSearch(ActionEvent actionEvent) throws IOException {
+    if (!tbar_search.getText().isEmpty()) {
+      App.ex.setSearchField(tbar_search.getText());
     }
-
-    /**
-     * Method that changes stage to Search page. It reads the value of the search
-     * field and if not empty it is passed to dataexchange
-     * @param actionEvent
-     * @throws IOException
-     */
-    public void switchToSearch(ActionEvent actionEvent) throws IOException {
-        if (!tbar_search.getText().isEmpty()){
-            App.ex.setSearchField(tbar_search.getText());
-        }
-        App.setRoot("search");
+    App.setRoot("search");
+  }
+
+  /**
+   * Method that changes stage to Explore page
+   *
+   * @param actionEvent
+   * @throws IOException
+   */
+  public void switchToExplore(ActionEvent actionEvent) throws IOException {
+    App.setRoot("explore");
+  }
+
+  /**
+   * Method that changes stage to Albums page
+   *
+   * @param actionEvent
+   * @throws IOException
+   */
+  public void switchToAlbums(ActionEvent actionEvent) throws IOException {
+    App.setRoot("explore_albums");
+  }
+
+  /**
+   * Method that changes stage to Map page
+   *
+   * @param actionEvent
+   * @throws IOException
+   */
+  public void switchToMap(ActionEvent actionEvent) throws IOException {
+    App.setRoot("map");
+  }
+
+  /**
+   * Method that changes stage to Upload page
+   *
+   * @param actionEvent the mouse has done something
+   * @throws IOException this page does not exist
+   */
+  public void switchToUpload(ActionEvent actionEvent) throws IOException {
+    App.setRoot("upload");
+  }
+
+  /**
+   * Method that creates album based on the user input and checked images
+   *
+   * @param actionEvent
+   */
+  public void createAlbum(ActionEvent actionEvent) {
+    String title = album_title_field.getText();
+    String description = album_desc_field.getText();
+    String tags = album_tag_field.getText();
+    List<Tag> tagsToAdd = TagService.getTagsFromString(tags);
+    User user = ApplicationState.getCurrentUser();
+
+      if (description.isEmpty()) {
+          description = "No desripton";
+      }
+
+    List<Node> imageContainers = new ArrayList<>(fileContainer.getChildren());
+    List<String> checkedImagesId = new ArrayList<>();
+    //Each image and checkbox has an hbox container
+    imageContainers.forEach(hbox ->
+        ((HBox) hbox).getChildren().stream()
+            //Filter children that is a checked checkbox
+            .filter(child -> child instanceof CheckBox && ((CheckBox) child).isSelected())
+            //Adds all checked image id
+            .forEach(checked -> checkedImagesId.add(checked.getId()))
+    );
+
+
+    if (validateInpid()) {
+      //Find the users images and makes a filter on the checked images
+      List<Image> albumImages = imageService.getImageFromUser(user).stream().filter(image -> checkedImagesId.contains(image.getId().toString())).collect(Collectors.toList());
+
+      if (albumImages.size() > 0) {
+        Optional<Album> createdAlbum = albumService.createAlbum(title, description, user, tagsToAdd, albumImages);
+        createdAlbum.ifPresent(album -> {
+          App.ex.setChosenAlbumId(album.getId());
+          try {
+            App.setRoot("main");
+          } catch (IOException e) {
+            e.printStackTrace();
+          }
+        });
+      } else {
+        Optional<Album> createdAlbum = albumService.createEmptyAlbum(title, description, user, tagsToAdd);
+        createdAlbum.ifPresent(album -> {
+          App.ex.setChosenAlbumId(album.getId());
+          try {
+            App.setRoot("main");
+          } catch (IOException e) {
+            e.printStackTrace();
+          }
+        });
+      }
     }
-
-    /**
-     * Method that changes stage to Explore page
-     * @param actionEvent
-     * @throws IOException
-     */
-    public void switchToExplore(ActionEvent actionEvent) throws IOException {
-        App.setRoot("explore");
-    }
-
-    /**
-     * Method that changes stage to Albums page
-     * @param actionEvent
-     * @throws IOException
-     */
-    public void switchToAlbums(ActionEvent actionEvent) throws IOException {
-        App.setRoot("explore_albums");
-    }
-
-    /**
-     * Method that changes stage to Map page
-     * @param actionEvent
-     * @throws IOException
-     */
-    public void switchToMap(ActionEvent actionEvent) throws IOException {
-        App.setRoot("map");
-    }
-
-    /**
-     * Method that changes stage to Upload page
-     * @param actionEvent the mouse has done something
-     * @throws IOException this page does not exist
-     */
-    public void switchToUpload(ActionEvent actionEvent) throws IOException {
-        App.setRoot("upload");
-    }
-
-    /**
-     * Method that creates album based on the user input and checked images
-     * @param actionEvent
-     */
-    public void createAlbum(ActionEvent actionEvent){
-        String title =  album_title_field.getText();
-        String description = album_desc_field.getText();
-        String tags = album_tag_field.getText();
-        List<Tag> tagsToAdd = TagService.getTagsFromString(tags);
-        User user = ApplicationState.getCurrentUser();
-
-        List<Node> imageContainers = new ArrayList<>(fileContainer.getChildren());
-        List<String> checkedImagesId = new ArrayList<>();
-        //Each image and checkbox has an hbox container
-        imageContainers.forEach(hbox ->
-                ((HBox) hbox).getChildren().stream()
-                        //Filter children that is a checked checkbox
-                        .filter(child -> child instanceof CheckBox && ((CheckBox) child).isSelected())
-                        //Adds all checked image id
-                        .forEach(checked -> checkedImagesId.add(checked.getId()))
-        );
-
-        //Find the users images and makes a filter on the checked images
-        List<Image> albumImages = imageService.getImageFromUser(user).stream().filter(image -> checkedImagesId.contains(image.getId().toString())).collect(Collectors.toList());
-
-        if (albumImages.size() > 0){
-            Optional<Album> createdAlbum = albumService.createAlbum(title, description, user, tagsToAdd, albumImages);
-            createdAlbum.ifPresent(album -> {
-                App.ex.setChosenAlbumId(album.getId());
-                try {
-                    App.setRoot("view_album");
-                } catch (IOException e) {
-                    e.printStackTrace();
-                }
-            });
-        }
-        else {
-            Optional<Album> createdAlbum = albumService.createEmptyAlbum(title, description, user, tagsToAdd);
-            createdAlbum.ifPresent(album -> {
-                App.ex.setChosenAlbumId(album.getId());
-                try {
-                    App.setRoot("view_album");
-                } catch (IOException e) {
-                    e.printStackTrace();
-                }
-            });
-        }
+  }
+
+  private boolean validateInpid() {
+    boolean check = true;
+    if (album_title_field.getText().isEmpty()) {
+      album_title_field.clear();
+      album_title_field.setStyle("-fx-prompt-text-fill: red");
+      album_title_field.setPromptText("* Please enter a title for the album");
+      check = false;
     }
+    return check;
+  }
 }
diff --git a/src/main/java/NTNU/IDATT1002/controllers/ViewAlbum.java b/src/main/java/NTNU/IDATT1002/controllers/ViewAlbum.java
index 65513152d5e6187e1dfdfe4890d5300d00b85034..0efaf351e8a56a645fb55cd6abb0c99667902a7d 100644
--- a/src/main/java/NTNU/IDATT1002/controllers/ViewAlbum.java
+++ b/src/main/java/NTNU/IDATT1002/controllers/ViewAlbum.java
@@ -82,9 +82,10 @@ public class ViewAlbum implements Initializable {
         Long currentAlbumId = App.ex.getChosenAlbumId();
 
         Optional<Album> foundAlbum = albumService.getAlbumById(currentAlbumId);
-        foundAlbum.ifPresent(album -> {
+        foundAlbum. ifPresent(album -> {
             currentAlbum = album;
-            NTNU.IDATT1002.models.Image titleImage = album.getImages().get(0);
+            NTNU.IDATT1002.models.Image titleImage = album.getImages().
+                get(0);
             Image image = ImageUtil.convertToFXImage(titleImage);
             mainPicture.setImage(image);
             pictureTitleField.setText("LEGG TIL BILDETITTEL HER");